home *** CD-ROM | disk | FTP | other *** search
/ EuroCD 3 / EuroCD 3.iso / Programming / Python-1.4 / Demo / classes / Rat.py < prev    next >
Text File  |  1998-06-24  |  2KB  |  104 lines
  1. # Rational numbers
  2.  
  3. from types import *
  4.  
  5. def rat(num, den):
  6.     if type(num) == FloatType or type(den) == FloatType:
  7.         return num/den
  8.     return Rat(num, den)
  9.  
  10.  
  11. def gcd(a, b):
  12.     while b:
  13.         a, b = b, a%b
  14.     return a
  15.  
  16.  
  17. class Rat:
  18.  
  19.     def __init__(self, num, den):
  20.         if den == 0:
  21.             raise ZeroDivisionError, 'rat(x, 0)'
  22.         if type(den) == FloatType or type(num) == FloatType:
  23.             g = float(den)
  24.         else:
  25.             g = gcd(num, den)
  26.         self.num = num/g
  27.         self.den = den/g
  28.  
  29.     def __repr__(self):
  30.         return 'Rat(%s, %s)' % (self.num, self.den)
  31.  
  32.     def __str__(self):
  33.         if self.den == 1:
  34.             return str(self.num)
  35.         else:
  36.             return '%s/%s' % (self.num, self.den)
  37.  
  38.     def __cmp__(a, b):
  39.         c = a-b
  40.         if c.num < 0:
  41.             return -1
  42.         if c.num > 0:
  43.             return 1
  44.         return 0
  45.  
  46.     def __float__(self):
  47.         return float(self.num) / float(self.den)
  48.  
  49.     def __long__(self):
  50.         return long(self.num) / long(self.den)
  51.  
  52.     def __int__(self):
  53.         return int(self.num / self.den)
  54.  
  55.     def __coerce__(a, b):
  56.         t = type(b)
  57.         if t == IntType:
  58.             return a, Rat(b, 1)
  59.         if t == LongType:
  60.             return a, Rat(b, 1L)
  61.         if t == FloatType:
  62.             return a, Rat(b, 1.0)
  63.         if t == InstanceType and a.__class__ == b.__class__:
  64.             return a, b
  65.         raise TypeError, 'Rat.__coerce__: bad other arg'
  66.  
  67.     def __add__(a, b):
  68.         return rat(a.num*b.den + b.num*a.den, a.den*b.den)
  69.  
  70.     def __sub__(a, b):
  71.         return rat(a.num*b.den - b.num*a.den, a.den*b.den)
  72.  
  73.     def __mul__(a, b):
  74.         return rat(a.num*b.num, a.den*b.den)
  75.  
  76.     def __div__(a, b):
  77.         return rat(a.num*b.den, a.den*b.num)
  78.  
  79.     def __neg__(self):
  80.         return rat(-self.num, self.den)
  81.  
  82.  
  83. def test():
  84.     print Rat(-1L, 1)
  85.     print Rat(1, -1)
  86.     a = Rat(1, 10)
  87.     print int(a), long(a), float(a)
  88.     b = Rat(2, 5)
  89.     l = [a+b, a-b, a*b, a/b]
  90.     print l
  91.     l.sort()
  92.     print l
  93.     print Rat(0, 1)
  94.     print a+1
  95.     print a+1L
  96.     print a+1.0
  97.     try:
  98.         print Rat(1, 0)
  99.         raise SystemError, 'should have been ZeroDivisionError'
  100.     except ZeroDivisionError:
  101.         print 'OK'
  102.  
  103. test()
  104.